package LeetCode.LC102;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * @author corn
 * @create 2020-08-11-21:32
 */

/**
 * 102. 二叉树的层序遍历
 *
 * 给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。
 */
public class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> lists = new ArrayList<>();
        List<Integer> integers = new ArrayList<>();
        //存储同一层的结点
        LinkedList<TreeNode> linkTree = new LinkedList<>();
        //判空
        if(root == null){
            return lists;
        }
        //添加根结点
        linkTree.add(root);
        //now代表当前层存在的结点个数。next：计算下一层添加的节点数
        int now = 1,next = 0;
        while (!linkTree.isEmpty()){
            //将当前结点从linkCode提取出来
            TreeNode t = linkTree.remove();
            now--;
            integers.add(t.val);
            //将当前结点的左右结点连接到后面
            if(t.left != null){
                linkTree.add(t.left);
                next++;
            }
            if(t.right != null){
                linkTree.add(t.right);
                next++;
            }
            //当now为0，即当前一层结点全部被提取出来
            if(now == 0){
                //将当前层数据添加到list中
                lists.add(new ArrayList<>(integers));
                //tmp清空
                integers.clear();
                //得到下一层的结点个数
                now = next;
                //重新再计算下一层结点个数
                next = 0;
            }
        }
        return lists;
    }
}
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
